package com.michael.leetcode;

import lombok.extern.slf4j.Slf4j;
import org.junit.Test;

import java.util.Stack;

@Slf4j
public class DeleteNode_450 {
    class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode() {
        }

        TreeNode(int val) {
            this.val = val;
        }

        TreeNode(int val, TreeNode left, TreeNode right) {
            this.val = val;
            this.left = left;
            this.right = right;
        }
    }

    /**
     * 450. 删除二叉搜索树中的节点
     * 给定一个二叉搜索树的根节点 root 和一个值 key，删除二叉搜索树中的 key 对应的节点，并保证二叉搜索树的性质不变。返回二叉搜索树（有可能被更新）的根节点的引用。
    一般来说，删除节点可分为两个步骤：
    首先找到需要删除的节点；
    如果找到了，删除它。

    示例 1:
    输入：root = [5,3,6,2,4,null,7], key = 3
    输出：[5,4,6,2,null,null,7]
    解释：给定需要删除的节点值是 3，所以我们首先找到 3 这个节点，然后删除它。
    一个正确的答案是 [5,4,6,2,null,null,7], 如下图所示。
    另一个正确答案是 [5,2,6,null,4,null,7]。

    示例 2:
    输入: root = [5,3,6,2,4,null,7], key = 0
    输出: [5,3,6,2,4,null,7]
    解释: 二叉树不包含值为 0 的节点

    示例 3:
    输入: root = [], key = 0
    输出: []


    提示:
    节点数的范围 [0, 104].
    -105 <= Node.val <= 105
    节点值唯一
    root 是合法的二叉搜索树
    -105 <= key <= 105

    进阶： 要求算法时间复杂度为 O(h)，h 为树的高度。
     */
    public TreeNode deleteNode(TreeNode root, int key) {
        if (root == null){
            return root;
        }

        if (root.val > key){
            root.left = deleteNode(root.left, key);
            return root;
        }else if (root.val < key){
            root.right = deleteNode(root.right, key);
            return root;
        }else if (root.val == key) {
            if (root.left == null && root.right == null) {
                return null;
            }
            if (root.right == null) {
                return root.left;
            }
            if (root.left == null) {
                return root.right;
            }
            TreeNode successor = root.right;
            while (successor.left != null) {
                successor = successor.left;
            }
            root.right = deleteNode(root.right, successor.val);
            successor.right = root.right;
            successor.left = root.left;
            return successor;
        }
        return root;
    }

    // 深度优先遍历
    public void searchTreeDeep(TreeNode root){
        Stack<TreeNode> stack = new Stack<>();
        stack.push(root);
        while (!stack.isEmpty()){
            TreeNode pop = stack.pop();
            log.info("pop {}", pop.val);
            if (pop.right != null){
                stack.push(pop.right);
            }

            if (pop.left != null){
                stack.push(pop.left);
            }
        }
    }

    @Test
    public void test(){
        TreeNode node2 = new TreeNode(2);
        TreeNode node4 = new TreeNode(4);
        TreeNode node7 = new TreeNode(7);
        TreeNode node3 = new TreeNode(3, node2, node4);
        TreeNode node6 = new TreeNode(6, null, node7);
        TreeNode root = new TreeNode(5, node3, node6);
        deleteNode(root,3);
    }
}